# ! usr/bin/python
# coding=utf-8   //这句是使用utf8编码方式方法， 可以单独加入python头使用。
# -*- coding: UTF-8 -*-
import sys
import zlib
import json
import math

sys.path.append("../pythonLib/")
import urllib.request
import re
import pymysql


from selenium import webdriver
from urllib import parse
from selenium.webdriver.common.by import By
from selenium.webdriver.support import expected_conditions as EC
from selenium.webdriver.common.action_chains import ActionChains
from selenium.webdriver.common.keys import Keys
from selenium.webdriver.common.desired_capabilities import DesiredCapabilities
from selenium.webdriver.common.proxy import ProxyType
from selenium.webdriver.support.ui import WebDriverWait
from selenium.webdriver.firefox.firefox_binary import FirefoxBinary
binary = FirefoxBinary(r"C:\Program Files (x86)\Mozilla Firefox\firefox.exe")
import selenium.webdriver.support.ui as ui
import random
import datetime
import time

import subprocess

conn = pymysql.connect(host='127.0.0.1', port=3306, user='root', passwd='159456', db='longzs',charset='utf8')
mysql = conn.cursor()


# mysql.close()
# conn.close()

# subprocess.call("php /path/to/my.php")
# 如果要返回值:
# proc = subprocess.Popen("php /path/to/my.php", shell=True,stdout=subprocess.PIPE)
# script_response = proc.stdout.read()

pcUserAgent = {
    "safari 5.1 – MAC": "User-Agent:Mozilla/5.0 (Macintosh; U; Intel Mac OS X 10_6_8; en-us) AppleWebKit/534.50 (KHTML, like Gecko) Version/5.1 Safari/534.50",
    "safari 5.1 – Windows": "User-Agent:Mozilla/5.0 (Windows; U; Windows NT 6.1; en-us) AppleWebKit/534.50 (KHTML, like Gecko) Version/5.1 Safari/534.50",
    "IE 9.0": "User-Agent:Mozilla/5.0 (compatible; MSIE 9.0; Windows NT 6.1; Trident/5.0);",
    "IE 8.0": "User-Agent:Mozilla/4.0 (compatible; MSIE 8.0; Windows NT 6.0; Trident/4.0)",
    "IE 7.0": "User-Agent:Mozilla/4.0 (compatible; MSIE 7.0; Windows NT 6.0)",
    "IE 6.0": "User-Agent: Mozilla/4.0 (compatible; MSIE 6.0; Windows NT 5.1)",
    "Firefox 4.0.1 – MAC": "User-Agent: Mozilla/5.0 (Macintosh; Intel Mac OS X 10.6; rv:2.0.1) Gecko/20100101 Firefox/4.0.1",
    "Firefox 4.0.1 – Windows": "User-Agent:Mozilla/5.0 (Windows NT 6.1; rv:2.0.1) Gecko/20100101 Firefox/4.0.1",
    "Opera 11.11 – MAC": "User-Agent:Opera/9.80 (Macintosh; Intel Mac OS X 10.6.8; U; en) Presto/2.8.131 Version/11.11",
    "Opera 11.11 – Windows": "User-Agent:Opera/9.80 (Windows NT 6.1; U; en) Presto/2.8.131 Version/11.11",
    "Chrome 17.0 – MAC": "User-Agent: Mozilla/5.0 (Macintosh; Intel Mac OS X 10_7_0) AppleWebKit/535.11 (KHTML, like Gecko) Chrome/17.0.963.56 Safari/535.11",
    "Maxthon": "User-Agent: Mozilla/4.0 (compatible; MSIE 7.0; Windows NT 5.1; Maxthon 2.0)",
    "Tencent TT": "User-Agent: Mozilla/4.0 (compatible; MSIE 7.0; Windows NT 5.1; TencentTraveler 4.0)",
    "The World 2.x": "User-Agent: Mozilla/4.0 (compatible; MSIE 7.0; Windows NT 5.1)",
    "The World 3.x": "User-Agent: Mozilla/4.0 (compatible; MSIE 7.0; Windows NT 5.1; The World)",
    "sogou 1.x": "User-Agent: Mozilla/4.0 (compatible; MSIE 7.0; Windows NT 5.1; Trident/4.0; SE 2.X MetaSr 1.0; SE 2.X MetaSr 1.0; .NET CLR 2.0.50727; SE 2.X MetaSr 1.0)",
    "360": "User-Agent: Mozilla/4.0 (compatible; MSIE 7.0; Windows NT 5.1; 360SE)",
    "Avant": "User-Agent: Mozilla/4.0 (compatible; MSIE 7.0; Windows NT 5.1; Avant Browser)",
    "Green Browser": "User-Agent: Mozilla/4.0 (compatible; MSIE 7.0; Windows NT 5.1)",
    "Chrome61 - Windows": "User-Agent:Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/61.0.3163.100 Safari/537.36"
}

mobileUserAgent = {
    "iOS 4.33 – iPhone": "User-Agent:Mozilla/5.0 (iPhone; U; CPU iPhone OS 4_3_3 like Mac OS X; en-us) AppleWebKit/533.17.9 (KHTML, like Gecko) Version/5.0.2 Mobile/8J2 Safari/6533.18.5",
    "iOS 4.33 – iPod Touch": "User-Agent:Mozilla/5.0 (iPod; U; CPU iPhone OS 4_3_3 like Mac OS X; en-us) AppleWebKit/533.17.9 (KHTML, like Gecko) Version/5.0.2 Mobile/8J2 Safari/6533.18.5",
    "iOS 4.33 – iPad": "User-Agent:Mozilla/5.0 (iPad; U; CPU OS 4_3_3 like Mac OS X; en-us) AppleWebKit/533.17.9 (KHTML, like Gecko) Version/5.0.2 Mobile/8J2 Safari/6533.18.5",
    "Android N1": "User-Agent: Mozilla/5.0 (Linux; U; Android 2.3.7; en-us; Nexus One Build/FRF91) AppleWebKit/533.1 (KHTML, like Gecko) Version/4.0 Mobile Safari/533.1",
    "Android QQ": "User-Agent: MQQBrowser/26 Mozilla/5.0 (Linux; U; Android 2.3.7; zh-cn; MB200 Build/GRJ22; CyanogenMod-7) AppleWebKit/533.1 (KHTML, like Gecko) Version/4.0 Mobile Safari/533.1",
    "Android Opera ": "User-Agent: Opera/9.80 (Android 2.3.4; Linux; Opera Mobi/build-1107180945; U; en-GB) Presto/2.8.149 Version/11.10",
    "Android Pad Moto Xoom": "User-Agent: Mozilla/5.0 (Linux; U; Android 3.0; en-us; Xoom Build/HRI39) AppleWebKit/534.13 (KHTML, like Gecko) Version/4.0 Safari/534.13",
    "BlackBerry": "User-Agent: Mozilla/5.0 (BlackBerry; U; BlackBerry 9800; en) AppleWebKit/534.1+ (KHTML, like Gecko) Version/6.0.0.337 Mobile Safari/534.1+",
    "WebOS HP Touchpad": "User-Agent: Mozilla/5.0 (hp-tablet; Linux; hpwOS/3.0.0; U; en-US) AppleWebKit/534.6 (KHTML, like Gecko) wOSBrowser/233.70 Safari/534.6 TouchPad/1.0",
    "Nokia N97": "User-Agent: Mozilla/5.0 (SymbianOS/9.4; Series60/5.0 NokiaN97-1/20.0.019; Profile/MIDP-2.1 Configuration/CLDC-1.1) AppleWebKit/525 (KHTML, like Gecko) BrowserNG/7.1.18124",
    "Windows Phone Mango": "User-Agent: Mozilla/5.0 (compatible; MSIE 9.0; Windows Phone OS 7.5; Trident/5.0; IEMobile/9.0; HTC; Titan)",
    "UC": "User-Agent: UCWEB7.0.2.37/28/999",
    "UC standard": "User-Agent: NOKIA5700/ UCWEB7.0.2.37/28/999",
    "UCOpenwave": "User-Agent: Openwave/ UCWEB7.0.2.37/28/999",
    "UC Opera": "User-Agent: Mozilla/4.0 (compatible; MSIE 6.0; ) Opera/UCWEB7.0.2.37/28/999"
}


def getHtml(url):
    page = urllib.request.urlopen(url)
    # decompressed_data = zlib.decompress(mybytes, 16 + zlib.MAX_WBITS)
    # text = decompressed_data.decode('utf8')
    html = page.read()  # .decode('utf8', 'ignore')  #.decode('GBK', 'ignore')
    return html


def filenr(path):
    file_object = open(path, 'r',encoding='utf-8')
    try:
        all_the_text = file_object.read()
    finally:
        file_object.close()
    return all_the_text

def writefilenr(path,nr):
    file_object = open(path, 'w',encoding='utf-8')
    file_object.write(nr)
    file_object.close()

def selectdriver(n):
    if n==1:
        options = webdriver.ChromeOptions()
        options.add_experimental_option("excludeSwitches", ["ignore-certificate-errors"])
        options.add_argument('--user-data-dir=C:/Users/yzs/AppData/Local/Google/Chrome/User Data/Default')  # 设置成用户自己的数据目录
        driver = webdriver.Chrome(chrome_options=options)
    elif n==2:
        dirver = webdriver.Firefox()
    elif n==3:
        driver = webdriver.PhantomJS()
        desired_capabilities = DesiredCapabilities.PHANTOMJS.copy()
        desired_capabilities["phantomjs.page.settings.userAgent"] = "User-Agent:Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/61.0.3163.100 Safari/537.36"
        # desired_capabilities["phantomjs.page.settings.loadImages"] = False # 不载入图片，爬页面速度会快很多
        driver.start_session(desired_capabilities)
    return driver



def updatecacthcield(name,value,url):
    global conn,mysql
    mysql.execute("select id from  longzs.`cache` where url='{}'".format(url))
    data = mysql.fetchall()
    if(len(data)!=0):
        mysql.execute("update longzs.`cache` set `{}`='{}' where url='{}'".format(name,pymysql.escape_string(value), url))
        conn.commit()
        r=data[0][0]
    else:
        # today = math.ceil(time.time()) #(datetime.datetime.now() + datetime.timedelta(days=0)).strftime("%Y-%m-%d %H:%M:%S")
        today =(datetime.datetime.now()).strftime("%Y-%m-%d %H:%M:%S") # + datetime.timedelta(hours=8)
        sql="insert into longzs.`cache` ("+name+",sj,url) values (%s,%s,%s)"
        value=(pymysql.escape_string(value),today,url)
        mysql.execute(sql,value)
        conn.commit()
        r=conn.insert_id()
    return r


